from lib.Character.Base import Base
from lib.Skill import EffectType
from lib.Skill.Effect import RoundEffect


class Dot(RoundEffect):
    def __init__(self, name, dot_type, round, target_self=False):
        super(Dot, self).__init__(name, dot_type, target_self, round)


class DotBase(RoundEffect):
    def __init__(self, name, round, base, target_self=False):
        super(DotBase, self).__init__(name, EffectType.DOT_BASE, round, target_self)
        if not isinstance(base, Base):
            raise AttributeError('base must type Character.Base')
        self.base = base
